#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
简单系统测试脚本
测试真实数据获取和分析系统
"""

import sys
import os
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from real_data_fetcher import RealDataFetcher
from advanced_analysis_system import AdvancedAnalysisSystem

def test_simple_analysis():
    """测试简单分析流程"""
    print("🚀 开始简单分析测试...")
    
    # 创建数据获取器
    fetcher = RealDataFetcher()
    
    # 测试几只知名股票
    test_stocks = ['000001', '000002', '600036', '600519']
    
    print("📊 测试股票数据获取...")
    for stock_code in test_stocks:
        print(f"\n🔍 测试股票: {stock_code}")
        
        # 获取历史数据
        hist_data = fetcher.get_stock_data(stock_code, 5)
        if not hist_data.empty:
            print(f"   ✅ 历史数据: {len(hist_data)} 条记录")
            print(f"   📈 最新价格: {hist_data['close'].iloc[-1]:.2f}元")
        else:
            print(f"   ❌ 历史数据获取失败")
            
        # 获取基本信息
        basic_info = fetcher.get_stock_basic_info(stock_code)
        if basic_info:
            print(f"   ✅ 基本信息获取成功")
            print(f"   💰 当前价格: {basic_info.get('current_price', 0):.2f}元")
        else:
            print(f"   ❌ 基本信息获取失败")
    
    print("\n" + "="*50)
    print("🎯 测试分析系统...")
    
    # 创建分析系统
    analysis_system = AdvancedAnalysisSystem()
    
    # 模拟候选股票
    mock_candidates = [
        {'code': '000001', 'name': '平安银行'},
        {'code': '000002', 'name': '万科A'},
        {'code': '600036', 'name': '招商银行'}
    ]
    
    print(f"📝 分析候选股票: {len(mock_candidates)} 只")
    
    # 测试深度分析
    try:
        results = analysis_system.analyze_candidates(mock_candidates)
        print(f"✅ 深度分析完成: {len(results)} 只股票成功分析")
        
        if results:
            print("📊 分析结果预览:")
            for result in results[:2]:  # 只显示前2个
                print(f"   • {result.get('code')}: 评分 {result.get('score', 0):.1f}")
        
    except Exception as e:
        print(f"❌ 深度分析异常: {e}")
    
    print("\n" + "="*50)
    print("🎉 测试完成！")

if __name__ == "__main__":
    test_simple_analysis()
